home *** CD-ROM | disk | FTP | other *** search
/ PCGUIA 127 / PC Guia 127.iso / Software / Utils / FlashGot / Bin / flashgot-0.5.9.993-fx+fl+mz+ns+tb.xpi / components / flashgotService.js < prev   
Text File  |  2006-02-14  |  73KB  |  2,366 lines
  1. /***** BEGIN LICENSE BLOCK *****
  2.  
  3.     FlashGot - a Firefox extension for external download managers integration
  4.     Copyright (C) 2004-2006 Giorgio Maone - g.maone@informaction.com
  5.  
  6.     This program is free software; you can redistribute it and/or modify
  7.     it under the terms of the GNU General Public License as published by
  8.     the Free Software Foundation; either version 2 of the License, or
  9.     (at your option) any later version.
  10.  
  11.     This program is distributed in the hope that it will be useful,
  12.     but WITHOUT ANY WARRANTY; without even the implied warranty of
  13.     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14.     GNU General Public License for more details.
  15.  
  16.     You should have received a copy of the GNU General Public License
  17.     along with this program; if not, write to the Free Software
  18.     Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  19.  
  20. ***** END LICENSE BLOCK *****/
  21.  
  22. // *****************************************************************************
  23. // START DMS CLASSES
  24. // *****************************************************************************
  25.  
  26. const ASK_NEVER = [false, false, false];
  27.  
  28. // *** Base/Windows DMS ********************************************************
  29. function FlashGotDM(name) {
  30.   if(arguments.length>0) {
  31.     this._init(name);
  32.   }
  33. }
  34.  
  35. FlashGotDM.init=function(service) {
  36.   FlashGotDM.cleanup();
  37.   FlashGotDM.dms=[];
  38.   FlashGotDM.dmtests={};
  39.   FlashGotDM.executables={};
  40.   FlashGotDM.initDMS(service);
  41. };
  42.  
  43. FlashGotDM.cleanup=function() {
  44.   if(! ("executables" in FlashGotDM) ) return;
  45.   var name;
  46.   for(name in FlashGotDM.executables) {
  47.     var f=FlashGotDM.executables[name];
  48.     if(f instanceof Components.interfaces.nsIFile) {
  49.       try { f.remove(true); } catch(ex) {}
  50.     }
  51.   }
  52. };
  53.  
  54. FlashGotDM.prototype = {
  55.   _init: function(name) {
  56.     this.name=name;
  57.     const dms=FlashGotDM.dms;
  58.     var pos=dms.length;
  59.     if(name in dms) {
  60.       var other=dms[name];
  61.       for(var j=pos; j-->0;) {
  62.         if(dms[j]==other) {
  63.           pos=j;
  64.           break;
  65.         }
  66.       }
  67.     }
  68.     dms[name]=dms[pos]=this;
  69.   }
  70. ,
  71.   _service: null,
  72.   _cookieManager: null,
  73.   _exeFile: false,
  74.   _supported: null,
  75.   custom: false,
  76.   disabledLink: false,
  77.   disabledSel: false,
  78.   disabledAll: false,
  79.   exeName: "FlashGot.exe",
  80.   askPath: ASK_NEVER,
  81.   cookieSupport: true,
  82.   postSupport: false
  83. ,  
  84.   get codeName() {
  85.     return this.name.replace(/\W/g,"_");
  86.   }
  87.  
  88. ,
  89.   get service() {
  90.     return this._service?this._service:this._service=
  91.     Components.classes[SERVICE_CTRID].getService(Components.interfaces.nsISupports).wrappedJSObject;
  92.   }
  93. ,
  94.   get cookieManager() {
  95.     return this._cookieManager?this._cookieManager:this._cookieManager=
  96.       Components.classes["@mozilla.org/cookiemanager;1"
  97.         ].getService(Components.interfaces.nsICookieManager);
  98.   }
  99. ,
  100.   get exeFile() {
  101.     if(typeof(this._exeFile)=="object") return this._exeFile;
  102.     const exeName=this.exeName;
  103.     if(!exeName) return this._exeFile=null;
  104.     if(typeof(FlashGotDM.executables[exeName])=="object") {
  105.       return this._exeFile=FlashGotDM.executables[exeName];
  106.     }
  107.     try {
  108.       const exeFile=Components.classes["@mozilla.org/file/local;1"].createInstance(
  109.         Components.interfaces.nsILocalFile);
  110.       exeFile.initWithPath(this.service.globals.profDir.path);
  111.       exeFile.append(exeName);
  112.       if(exeFile.exists()) {
  113.         try { exeFile.remove(true); } catch(ex) { this.log(ex.message); }
  114.       }
  115.       this._exeFile=this.checkExePlatform(exeFile);
  116.       if(this._exeFile!=null && this.createExecutable()) {
  117.         this.log(this._exeFile.path+" created");
  118.       }
  119.     } catch(ex) {
  120.       this._exeFile=null;
  121.       this.log("Can't init "+exeName+":\n"+ex.message);
  122.     }
  123.     return FlashGotDM.executables[exeName]=this._exeFile;
  124.   }
  125. ,
  126.   checkExePlatform: function(exeFile) {
  127.     return /(\/.*\.exe)|(\\.*\.sh)$/i.test(exeFile.path)?null:exeFile;
  128.   }
  129. ,
  130.   get supported() {
  131.     if(typeof(this._supported)=="boolean") return this._supported;
  132.     if(this.customSupportCheck && this.customSupportCheck()) return true;
  133.     if(!this.exeName) return true;
  134.     if(!this.exeFile) return false;
  135.     
  136.     var dmtest;
  137.     if(typeof(FlashGotDM.dmtests[this.exeName])!="string") {
  138.       const dmtestFile=this.service.tmpDir.clone();
  139.       dmtestFile.append(this.exeName+".test");
  140.       try {
  141.         this.launchSupportTest(dmtestFile);
  142.         this.log(dmtest=this.service.readFile(dmtestFile)); 
  143.       } catch(ex) {
  144.         this.log(ex.message);
  145.         dmtest="";
  146.       }
  147.       FlashGotDM.dmtests[this.exeName]=dmtest;
  148.     } else dmtest=FlashGotDM.dmtests[this.exeName];
  149.     return this._supported=dmtest.indexOf(this.name+"|OK")>-1;
  150.   }
  151. ,
  152.   launchSupportTest: function (testFile) {
  153.     this.runNative(["-o",testFile.path],true);
  154.   },
  155.   
  156.   shouldList: function() {
  157.     return this.supported;
  158.   }
  159. ,
  160.   log: function(msg) {
  161.     this.service.log(msg);
  162.   }
  163. ,
  164.   updateProgress: function(links, idx , len) {
  165.    
  166.     if((idx % 100) == 0) {
  167.       if(!len) {
  168.         links.progress.update(100);
  169.         return;
  170.       }
  171.       links.progress.update(50 + 49 * idx / len);
  172.     }
  173.     
  174.   }
  175. ,
  176.   isValidLink: null
  177. ,
  178.   createJobHeader: function(links, opType) {
  179.     return links.length+";"+this.name+";"+
  180.       (this.service.getPref(this.codeName+".quiet."+opType,false)
  181.         ?this.service.OP_QET:opType)
  182.       +";"+links.folder+";\n"
  183.   }
  184. ,
  185.   createJobBody: function(links) {
  186.     var job="";
  187.     var l,url;
  188.     const len=links.length;
  189.     this.checkCookieSupport();
  190.     var postData = links.postData || "";
  191.  
  192.     for(var j=0; j<len; j++) {
  193.       job+="\n"+(url=(l=links[j]).href) + "\n" +
  194.            l.description + "\n" +
  195.            this.getCookie(l,links) + "\n"
  196.            + postData;
  197.       this.updateProgress(links,j,len);
  198.     }
  199.     return job;
  200.   }
  201. ,
  202.   createJob: function(links,opType) {
  203.     var job=this.createJobHeader(links,opType) 
  204.     + this.getReferrer(links)
  205.     + this.createJobBody(links)+"\n";
  206.      if(job.substring(job.length-1)!="\n") {
  207.       job+="\n";
  208.      }
  209.     if(typeof(links.document)=="object") {
  210.       job+= links.document.referrer+ "\n" +links.document.cookie;
  211.     } else {
  212.       job+="\n";
  213.     }
  214.     return job;
  215.   }
  216. ,
  217.   _bgJob: true,
  218.   get bgJob() {
  219.     return this._bgJob && this.service.bgProcessing
  220.       ; 
  221.   }
  222. ,
  223.   download: function(links, opType) {
  224.     try {
  225.       links.folder=(links.length>0)?this.selectFolder(links, opType):""; 
  226.       this.performJob(this.createJob(links,opType));
  227.     } catch(ex) {
  228.       this.log(ex.message);
  229.     } finally {
  230.       this.updateProgress(links, 0); // 100%
  231.     }
  232.   }
  233. ,
  234.   getReferrer: function(links) {
  235.     return this.service.getPref("autoReferrer",true) ?
  236.       (links.referrer || 
  237.         typeof(links.document)=="object" && links.document.URL ||
  238.         links[0] && links[0].href || 
  239.         "about:blank"
  240.       ) : this.service.getPref("fakeReferrer","");
  241.   }
  242. ,
  243.   checkCookieSupport: function() {
  244.     this.getCookie=this.cookieSupport && !this.service.getPref("omitCookies")
  245.     ?this._getCookie
  246.     :function() { return ""; }
  247.     ;
  248.   }
  249. ,
  250.   getCookie: function() { return ""; }
  251. ,
  252.   _getCookie: function(link,links) {
  253.     if(!this.cookieSupport) return (this.getCookie=function() { return ""; })();
  254.     
  255.     var host,cookies;
  256.     if(cookies=links.cookies) {
  257.       host=link.host;
  258.       if(host) {
  259.         var c=cookies[host];
  260.         return c?c:"";
  261.       }
  262.       return "";
  263.     }
  264.     
  265.     var j,objCookie;
  266.     const hostCookies={};
  267.     
  268.     var l,parts;
  269.     for(j=links.length; j-->0;) {
  270.       l=links[j];
  271.       parts=l.href.match(/http[s]{0,1}:\/\/([^\/]+\.[^\/]+)/i); // host?
  272.       if(parts) {
  273.         hostCookies[l.host=parts[1]]="";
  274.       } else {
  275.         l.host=null;
  276.       }
  277.     }
  278.     
  279.     var cookieHost,cookieTable,tmpCookie;
  280.     const domainCookies={};
  281.  
  282.     for(var iter = this.cookieManager.enumerator; iter.hasMoreElements();) {
  283.       if((objCookie=iter.getNext()) instanceof Components.interfaces.nsICookie) {
  284.         cookieHost=objCookie.host;
  285.         if(cookieHost.charAt(0)==".") {
  286.           cookieHost=cookieHost.substring(1);
  287.           cookieTable=domainCookies;
  288.           if(typeof(tmpCookie=domainCookies[cookieHost])!="string") {
  289.             tmpCookie="";
  290.           }
  291.         } else {
  292.           if(typeof(tmpCookie=hostCookies[cookieHost])!="string") continue;
  293.           cookieTable=hostCookies;
  294.         }
  295.         cookieTable[cookieHost]=tmpCookie.concat(objCookie.name+"="+objCookie.value+"; ");
  296.       }
  297.     }
  298.     
  299.    
  300.     for(cookieHost in hostCookies) {
  301.       var dotPos;
  302.       for(host=cookieHost; (dotPos=host.indexOf('.'))>=0; ) { 
  303.         if(tmpCookie=domainCookies[host]) {
  304.           hostCookies[cookieHost]+=tmpCookie;
  305.         }
  306.         host=host.substring(dotPos+1);
  307.       }
  308.     }
  309.     
  310.     links.cookies=hostCookies;
  311.     return this.getCookie(link,links);
  312.   }
  313. ,
  314.   createJobFile: function(job) {
  315.     const jobFile=this.service.tmpDir.clone();
  316.     jobFile.append("flashgot.fgt");
  317.     jobFile.createUnique(0,0700);
  318.     this.service.writeFile(jobFile, job);
  319.     return jobFile;
  320.   }
  321. , 
  322.   _waitForNative: true,
  323.   get waitForNative() {
  324.     return this._waitForNative && this.service.bgProcessing;
  325.   }
  326. ,
  327.   performJob: function(job) {
  328.     const jobFile=this.createJobFile(job);
  329.     this.runNative([jobFile.path],this.waitForNative);
  330.   }
  331. ,
  332.   createExecutable: function() {
  333.     const exeFile=this.exeFile;
  334.     if(!exeFile) return false;
  335.     
  336.     const cc=Components.classes;
  337.     const ci=Components.interfaces;
  338.     const ios=cc['@mozilla.org/network/io-service;1'].getService(ci.nsIIOService);
  339.     const bis=cc['@mozilla.org/binaryinputstream;1'].createInstance(ci.nsIBinaryInputStream);
  340.     
  341.     var channel;
  342.     bis.setInputStream((
  343.       channel=
  344.         ios.newChannel("chrome://flashgot/content/"+this.exeName,null,null)
  345.     ).open())
  346.     ;
  347.     const bytesCount=channel.contentLength;
  348.     
  349.     const os=cc["@mozilla.org/network/file-output-stream;1"].createInstance(
  350.       ci.nsIFileOutputStream);
  351.     
  352.     try {
  353.       
  354.       os.init(exeFile,0x02 | 0x08, 0700, 0);
  355.       const bos=cc['@mozilla.org/binaryoutputstream;1'].createInstance(ci.nsIBinaryOutputStream);
  356.       bos.setOutputStream(os);
  357.       bos.writeByteArray(bis.readByteArray(bytesCount),bytesCount);
  358.       bos.close();
  359.  
  360.     } catch(ioErr) { // locked?
  361.       try {
  362.         if(exeFile.exists()) { // security check: it must be the right exe!
  363.           const testBis=cc['@mozilla.org/binaryinputstream;1'].createInstance(
  364.             ci.nsIBinaryInputStream);
  365.           testBis.setInputStream(
  366.             (channel=ios.newChannelFromURI(ios.newFileURI(exeFile))).open());
  367.           const error=new Error("Old, corrupt or unlegitemately modified "
  368.             +exeFile.path
  369.             +".\nThe file is locked: please delete it manually\n");
  370.             +ioErr.message;
  371.           if(channel.contentLength!=bytesCount) throw error;
  372.          
  373.           const legitimateData=bis.readByteArray(bytesCount);
  374.           const testData=testBis.readByteArray(bytesCount);
  375.           for(var j=bytesCount; j-->0;) {
  376.             if(legitimateData[j]!=testData[j]) throw new error;
  377.           }
  378.         } else throw ioErr;
  379.       } catch(unrecoverableErr) {
  380.          this.log("Error creating native executable\n"+exeFile.path+"\n"+unrecoverableErr.message);
  381.       }
  382.     } finally {
  383.       os.close();
  384.       bis.close();
  385.     }
  386.     
  387.     return true;
  388.   }
  389. ,
  390.   runNative: function(args,blocking,exeFile) {
  391.     try {
  392.       if(typeof(exeFile)=="object"
  393.         || (exeFile=this.exeFile).exists()
  394.         || this.createExecutable()) {
  395.         const proc=Components.classes['@mozilla.org/process/util;1'].createInstance(
  396.           Components.interfaces.nsIProcess);
  397.         proc.init(exeFile);
  398.         this.log("Running "+exeFile.path+" ("+(blocking?"blocking":"async")+")");
  399.         proc.run(blocking,args,args.length,{});
  400.         if(blocking && proc.exitValue!=0) {
  401.           this.log("Warning: native invocation of\n"
  402.             +exeFile.path
  403.             +"\nwith arguments <"
  404.             +args.join(" ")
  405.             +">\nreturned "+proc.exitValue);
  406.         }
  407.         return proc.exitValue;
  408.       } else {
  409.         this.log("Bad executable "+exeFile);
  410.       }
  411.     } catch(err) {
  412.       this.log("Error running native executable:\n"+exeFile.path+" "+args.join(" ")+"\n"+err.message);
  413.     }  
  414.     return 0xffffffff;
  415.   }
  416. ,
  417.   getWindow: function() {
  418.     return this.service.getWindow();
  419.   }
  420. ,
  421.   selectFolder: function(links,opType) { 
  422.     const cc=Components.classes;
  423.     const ci=Components.interfaces;
  424.    
  425.     const autoPref_FF="browser.download.useDownloadDir";
  426.     const autoPref_Moz="browser.download.autoDownload";
  427.     
  428.     var initialDir=null;
  429.     var downloadDir=null;
  430.     links.quickDownload = false;
  431.     
  432.     const pref = cc["@mozilla.org/preferences-service;1"].getService(ci.nsIPrefBranch);
  433.     
  434.     function findDownloadDir(prefName) {
  435.       try {
  436.         downloadDir = initialDir = pref.getComplexValue("browser.download.dir", ci.nsILocalFile);
  437.         return prefName;
  438.       } catch(ex) {
  439.         return "";
  440.       }
  441.     }
  442.   
  443.     const downloadDirPref = findDownloadDir("browser.download.dir") ||
  444.                     findDownloadDir("browser.download.downloadDir") || 
  445.                     findDownloadDir("browser.download.defaultFolder") ||
  446.                     "browser.download.dir"; 
  447.    
  448.     try {
  449.       links.quickDownload = pref.getBoolPref(autoPref_FF);
  450.     } catch(noFFEx) {
  451.       try {
  452.         links.quickDownload=pref.getBoolPref(autoPref_Moz);
  453.       } catch(noMozEx) {}
  454.     }
  455.    
  456.     if(!this.askPath[opType]) return "";
  457.     
  458.     if(downloadDir && downloadDir.exists() && downloadDir.isDirectory()  && 
  459.         links.quickDownload) {
  460.       return downloadDir.path;
  461.     }
  462.     
  463.     var title;
  464.     try {
  465.       var bundle = cc["@mozilla.org/intl/stringbundle;1"].getService(ci.nsIStringBundleService);
  466.       bundle = bundle.createBundle("chrome://mozapps/locale/downloads/unknownContentType.properties");
  467.       title = bundle.GetStringFromName("myDownloads");
  468.     } catch(ex) {
  469.       title="Download directory";
  470.     }
  471.     title='FlashGot ('+this.name+') - '+title;
  472.     
  473.     const fp = cc["@mozilla.org/filepicker;1"].createInstance(ci.nsIFilePicker);
  474.     const win=this.getWindow();
  475.     fp.init(win, title, ci.nsIFilePicker.modeGetFolder);
  476.     try {
  477.       if (initialDir &&  initialDir.exists() && initialDir.isDirectory()) {
  478.         fp.displayDirectory = initialDir;
  479.       }
  480.     } catch (ex) { this.log(ex); }
  481.     
  482.     fp.appendFilters(ci.nsIFilePicker.filterAll);
  483.  
  484.     if (fp.show()==ci.nsIFilePicker.returnOK) {
  485.       var localFile = fp.file.QueryInterface(ci.nsILocalFile);
  486.       pref.setComplexValue(downloadDirPref, ci.nsILocalFile, localFile);
  487.       var path=new String(localFile.path);
  488.       path._fgSelected=true;
  489.       return path;
  490.     }
  491.     
  492.     throw new Error("Download cancelled by user");
  493.   }
  494.   
  495. }
  496.  
  497.  
  498.  
  499.  
  500. // *** Unix-like DMS ***********************************************************
  501. function FlashGotDMX(name,cmd,argsTemplate) {
  502.   if(arguments.length!=0) {
  503.     this._init(name);
  504.     const cmds=FlashGotDMX.prototype.unixCmds;
  505.     cmds[cmds.length] = {longName: name, shortName: cmd};
  506.     this.unixCmd = cmd;
  507.     if(argsTemplate) this.argsTemplate = argsTemplate;
  508.   }
  509. }
  510. FlashGotDMX.prototype=new FlashGotDM();
  511. FlashGotDMX.constructor=FlashGotDMX;
  512. FlashGotDMX.prototype.exeName="flashgot.sh";
  513. FlashGotDMX.prototype.cookieSupport=false;
  514. FlashGotDMX.prototype.askPath=[true,true,true];
  515. FlashGotDMX.prototype.unixCmds=[];
  516. FlashGotDMX.prototype.unixShell=null;
  517. FlashGotDMX.prototype.argsTemplate="[URL]";
  518. FlashGotDMX.prototype.launchSupportTest=function(testFile) {
  519.   const cmds=this.unixCmds;
  520.   var script="(\n";
  521.   var cmd;
  522.   for(var j=cmds.length; j-->0;) {
  523.     cmd=cmds[j];
  524.     script+=" [ -x \"`which '"+cmd.shortName+"'`\" ] && echo '"
  525.       +cmd.longName+"|OK' || echo '"+cmd.longName+"|KO'\n"; 
  526.   }
  527.   script+=") > '"+ testFile.path + "'\n"; 
  528.   this.performJob(script,true);
  529. };
  530.  
  531. FlashGotDMX.prototype.createCmdLine=function(URL,REFERER,COOKIE,FOLDER,POST) {
  532.   const parms={ URL: URL, REFERER: REFERER, COOKIE: COOKIE, FOLDER: FOLDER, POST: POST };
  533.   return this.unixCmd+ " " +
  534.     this.argsTemplate.replace(/\[(.*?)(URL|REFERER|COOKIE|FOLDER|POST)(.*?)\]/g,
  535.       function(all,before,parm,after) { 
  536.           v=parms[parm]; 
  537.           return typeof(v)!="undefined" && v!=null
  538.             ?before+v+after
  539.             :"";
  540.       }
  541.       ) +" &\n";
  542. };
  543. FlashGotDMX.prototype.shellEsc = function(s) {
  544.   return s?s.replace(/([\\\*\?\[\]\$&<>\|\(\)\{\};"'`])/g,"\\$1").replace(/\s/g,"\\ "):null;
  545. };
  546. FlashGotDMX.prototype.createJob=function(links,opType) {
  547.   const shellEsc = this.shellEsc;
  548.   // basic implementation
  549.   const len=links.length;
  550.   const folder=shellEsc(links.folder);
  551.   const referrer=shellEsc(this.getReferrer(links));
  552.   const postData=shellEsc(links.postData);
  553.   var job="";
  554.   var l,url;
  555.   this.checkCookieSupport();
  556.   for(var j=0; j<len; j++) {
  557.     l=links[j];
  558.     url=l.href;
  559.     job+=this.createCmdLine(
  560.       shellEsc(url), 
  561.       referrer, 
  562.       shellEsc(this.getCookie(l,links)), 
  563.       folder, 
  564.       postData);
  565.     this.updateProgress(links, j, len);
  566.   }
  567.   return job;
  568. };
  569. FlashGotDMX.prototype.performJob=function(job,blocking) {
  570.   const jobFile=this.createJobFile("#!"+this.unixShell.path+"\n"+job);
  571.   jobFile.permissions=0700;
  572.   this.runNative([],
  573.     this.waitForNative || (typeof(blocking)!="undefined" && blocking),
  574.     jobFile);
  575. };
  576. FlashGotDMX.prototype.checkExePlatform=function(exeFile) {
  577.   const f=Components.classes["@mozilla.org/file/local;1"].createInstance(
  578.     Components.interfaces.nsILocalFile);
  579.   try {
  580.     f.initWithPath("/bin/sh");
  581.     if(f.exists()) {
  582.       FlashGotDMX.prototype.unixShell=f;
  583.       return exeFile;
  584.     }
  585.     this.log(f.path+" not found");
  586.   } catch(ex) {
  587.     this.log(ex.message);
  588.   }
  589.   return null;
  590. };
  591. FlashGotDMX.prototype.createExecutable=function() {
  592.   return false;
  593. };
  594.  
  595.  
  596.  
  597. // *** Mac OS X DMS ************************************************************
  598. function FlashGotDMMac(name, creatorId, macAppName) {
  599.   if(arguments.length!=0) {
  600.     this._initMac(name,creatorId,macAppName);
  601.   }
  602. }
  603. FlashGotDMMac.prototype=new FlashGotDM();
  604. FlashGotDMMac.constructor=FlashGotDMMac;
  605. FlashGotDMMac.prototype.exeName="flashgot-mac.sh";
  606. FlashGotDMMac.prototype.cookieSupport=false;
  607. FlashGotDMMac.prototype.OSASCRIPT="/usr/bin/osascript";
  608. FlashGotDMMac.prototype.macCreators=[];
  609. FlashGotDMMac.prototype._initMac=function(name, creatorId, macAppName) {
  610.   this._init(name);
  611.   if(creatorId) {
  612.     const creators=FlashGotDMMac.prototype.macCreators;
  613.     creators[creators.length] = {name: name, id: creatorId};
  614.   }
  615.   this.macAppName = macAppName?macAppName:name;
  616. };
  617. FlashGotDMMac.prototype.createScriptLauncher=function(scriptPath) {
  618.   return "#!/bin/sh\n"
  619.     +this.OSASCRIPT+" '"+scriptPath+"'";
  620. };
  621. FlashGotDMMac.prototype.checkExePlatform=function(exeFile) {
  622.   const f=Components.classes["@mozilla.org/file/local;1"].createInstance(
  623.     Components.interfaces.nsILocalFile);
  624.   try {
  625.     f.initWithPath(this.OSASCRIPT);
  626.     if(f.exists()) return exeFile;
  627.     this.log(f.path+" not found");
  628.   } catch(ex) {
  629.     this.log(ex.message);
  630.   }
  631.   return null;
  632. };
  633. FlashGotDMMac.prototype.createExecutable=function() {
  634.   const exeFile=this.exeFile;
  635.   if(exeFile) {
  636.     try {
  637.      const script=this.service.tmpDir.clone();
  638.      script.append("flashgot-test.scpt");
  639.      FlashGotDMMac.prototype.testAppleScript=script;
  640.      script.createUnique(0,0700);
  641.      if(exeFile.exists()) exeFile.remove(true);
  642.      exeFile.create(0,0700);
  643.      this.service.writeFile(exeFile, this.createScriptLauncher(script.path));
  644.      exeFile.permissions=0700;
  645.      return true;
  646.     } catch(ex) {
  647.       this.log(ex.message);
  648.     }
  649.   }
  650.   return false;
  651. };
  652. FlashGotDMMac.prototype.launchSupportTest=function(testFile) {
  653.   const creators=FlashGotDMMac.prototype.macCreators;
  654.   const RESP="    do shell script \"echo >>'"+testFile.path+"' '\" & theName & \"|";
  655.   function response(msg) {
  656.     return RESP+msg+"'\"\n";
  657.   }
  658.   var s="on test(theName, theCreator)\n"
  659.        +" tell app \"Finder\"\n"
  660.        +"  set theResponse to \"KO\"\n"
  661.        +"  try\n"
  662.        +"   get name of application file id theCreator\n"
  663.        +"   if result contains theName then\n"
  664.        +"     set theResponse to \"OK\"\n"
  665.        +"   end if\n"
  666.        +"  on error\n"
  667.        +"  end try\n"
  668.        +"  do shell script \"echo >>'"+testFile.path+"' '\" & theName & \"|\" & theResponse & \"'\"\n"
  669.        +" end tell\n"
  670.        +"end test\n"
  671.        +"\n";
  672.   for(var j=creators.length; j-->0;) {
  673.      s+='get test("'+creators[j].name+'","'+creators[j].id+'")\n'; 
  674.    }
  675.    this.service.writeFile(this.testAppleScript,s);
  676.    this.runNative([],true,this.exeFile);
  677. };
  678. FlashGotDMMac.prototype.performJob=function(job) {
  679.   const script=this.createJobFile(job);
  680.   const launcher=this.createJobFile(this.createScriptLauncher(script.path));
  681.   launcher.permissions=0700;
  682.   this.runNative([],this.waitForNative,launcher);
  683. };
  684. FlashGotDMMac.prototype.createJob=function(links,opType) {
  685.   const referrer=this.getReferrer(links);
  686.   var job = "tell application \""+ this.macAppName+ "\"\n";
  687.   for(var j=0,len=links.length; j<len; j++) {
  688.     job+="GetURL \""+links[j].href+"\" from \""+ referrer  +"\"\n";
  689.     this.updateProgress(links, j, len);
  690.   }
  691.   job+="end tell\n";
  692.   return job;
  693. };
  694.  
  695.  
  696.  
  697. // *** Custom DMS **************************************************************
  698. function FlashGotDMCust(name) {
  699.   if(arguments.length==0 || (!name) || (!name.length)) return;
  700.   name=name.replace(/,/g," ");
  701.   this._init(name);
  702.   
  703.   const cc=Components.classes;
  704.   const ci=Components.interfaces;
  705.   const svc=this.service;
  706.   this.prefsBase="custom."+this.codeName+".";
  707. }
  708.  
  709. FlashGotDMCust.init=function(service) {
  710.   const names=service.getPref("custom","").split(/\s*,\s*/);
  711.   for(var j=names.length; j-->0;) {
  712.     new FlashGotDMCust(names[j]);
  713.   }
  714. }
  715.  
  716. FlashGotDMCust.persist=function(service) {
  717.   const dms=FlashGotDM.dms;
  718.   const cdms=[];
  719.   for(var j=dms.length; j-->0;) {
  720.     if(dms[j].custom) cdms.push(dms[j].name);
  721.   }
  722.   service.setPref("custom", cdms.join(","));
  723. }
  724.  
  725. FlashGotDMCust.prototype=new FlashGotDM();
  726. delete FlashGotDMCust.prototype.launchSupportTest;
  727. delete FlashGotDMCust.prototype.exeFile;
  728.  
  729. FlashGotDMCust.constructor=FlashGotDMCust;
  730.  
  731. FlashGotDMCust.prototype.custom = true;
  732. FlashGotDMCust.prototype._supported = true;
  733. FlashGotDMCust.prototype.__defineGetter__("exeFile",function() {
  734.   try {
  735.     return this.service.prefs.getComplexValue(this.prefsBase+"exe", 
  736.       Components.interfaces.nsILocalFile);
  737.   } catch(ex) {
  738.     return null;
  739.   }
  740. });
  741. FlashGotDMCust.prototype.__defineSetter__("exeFile",function(v) {
  742.   try {
  743.     if(v) {
  744.       this.service.prefs.setComplexValue(this.prefsBase+"exe", 
  745.           Components.interfaces.nsILocalFile,v);
  746.       return v;
  747.     }
  748.   } catch(ex) {
  749.     return null;
  750.   }
  751. });
  752.  
  753. FlashGotDMCust.prototype.__defineGetter__("argsTemplate",function() {
  754.   var t=this.service.getPref(this.prefsBase+"args","[URL]");
  755.   return /['"`]/.test(t)?this.argsTemplate=t:t;  
  756. });
  757. FlashGotDMCust.prototype.__defineSetter__("argsTemplate",function(v) {
  758.   if(!v) {
  759.     v="";
  760.   } else {
  761.     v=v.replace(/['"`]/g,"");
  762.   }
  763.   this.service.setPref(this.prefsBase+"args",v);
  764.   return v;
  765. });
  766.  
  767. FlashGotDMCust.prototype.cookieSupport=false;
  768. FlashGotDMCust.prototype.askPath=[true,true,true];
  769.  
  770. FlashGotDMCust.prototype.download = function(links,opType) {
  771.   const t=this.argsTemplate;
  772.   this.cookieSupport=/\[.*?COOKIE.*?\]/.test(t);
  773.   this.askPath[opType]=/\[.*?FOLDER.*?\]/.test(t);
  774.   var exeFile=this.exeFile;
  775.   // portable hacks
  776.   if(exeFile && !exeFile.exists()) {
  777.     // try changing the first part of path
  778.     var path = exeFile.path;
  779.     var profPath = this.service.profDir.path;
  780.     var pos1, pos2;
  781.     if(path[1] == ":" && profPath[1] == ":") { 
  782.       // easy, it's Windows, swap drive letter
  783.       path = profPath[0] + path.substring(1);
  784.     } else if(path.indexOf("/mount/") == 0 && profPath.indexOf("/mount/") == 0) {
  785.       pos1 = path.indexOf("/", 7);
  786.       pos2 = profPath.indexOf("/", 7);
  787.       path = "/mount/" + profPath.substring(7, pos2) + path.substring(pos1); 
  788.     } else if((pos1 = path.indexOf("/",1)) > 0 && (pos2 = profPath.indexOf("/", 1)) > 0) {
  789.       path = profPath.substring(0, pos2) + path.substring(pos1);
  790.     } else exeFile = null;
  791.     if(exeFile) {
  792.       exeFile = exeFile.clone().QueryInterface(Components.interfaces.nsILocalFile).initWithPath(path);
  793.       if(!exeFile.exists()) exeFile = null;
  794.     }
  795.   }
  796.   links.exeFile= (exeFile || 
  797.     (exeFile=this.exeFile=this.locateExeFile())) ? exeFile : null;
  798.   FlashGotDM.prototype.download.call(this, links, opType);
  799. };
  800.  
  801. FlashGotDMCust.prototype.locateExeFile = function(name) {
  802.   const cc=Components.classes;
  803.   const ci=Components.interfaces;
  804.   if(!name) name=this.name;
  805.   var title=this.service.getString("custom.exeFile");
  806.   title='FlashGot ('+name+') - '+title;
  807.   
  808.   const fp = cc["@mozilla.org/filepicker;1"].createInstance(ci.nsIFilePicker);
  809.   const win=this.getWindow();
  810.   fp.init(win, title, ci.nsIFilePicker.modeOpen);
  811.   fp.appendFilters(ci.nsIFilePicker.filterApps);
  812.   fp.appendFilters(ci.nsIFilePicker.filterAll);
  813.  
  814.   if (fp.show() == ci.nsIFilePicker.returnOK) {
  815.     var file = fp.file.QueryInterface(ci.nsILocalFile);
  816.     if(file.exists()) {
  817.       return file;
  818.     }
  819.   }
  820.   return null;
  821. };
  822. FlashGotDMCust.prototype.PLACEHOLDERS=["URL","REFERER","COOKIE","FOLDER","POST", "UFILE", "CFILE"];
  823. FlashGotDMCust.prototype.postSupport = true;
  824. FlashGotDMCust.prototype._addParts=function(a,s) {
  825.   var parts=s.split(/\s+/);
  826.   var k,p;
  827.   for(k in parts) {
  828.     if((p=parts[k]).length) {
  829.       a[a.length] = p;
  830.     }
  831.   } 
  832. };
  833. FlashGotDMCust.prototype.makeArgs=function(URL, REFERER, COOKIE, FOLDER, POST,  CFILE, UFILE) {
  834.   const parms={ URL: URL, REFERER: REFERER, COOKIE: COOKIE, FOLDER: FOLDER, POST: POST,  CFILE: CFILE, UFILE: UFILE};
  835.   const args=[];
  836.   var t=this.argsTemplate;
  837.   var j,v,len,s;
  838.  
  839.   var idx;
  840.   for(var m; 
  841.       m=t.match( /([\n.]*?)\[([\n.]*?)(\S*)(URL|REFERER|COOKIE|FOLDER|POST|CFILE|UFILE)([^\s\]]*)([\n.]*?)\]([\n.]*?)/); 
  842.       t=t.substring(idx+m[0].length) ) {
  843.     
  844.     if((idx=m.index)>0) {
  845.       this._addParts(args,t.substring(0,idx));
  846.     }
  847.     
  848.     v=parms[m[4]];
  849.     len=typeof(v) != "undefined" && v!=null?m.length:2;
  850.     for(j =1 ; j < len; j++) {
  851.       if(j == 3) {
  852.         args[args.length] = m[3] + v + m[5];
  853.         j = 5;
  854.       } else {
  855.         s=m[j];
  856.         if(s.length) {
  857.           this._addParts(args, s);
  858.         }
  859.       }
  860.     }
  861.   }
  862.   
  863.   if(t.length) {
  864.     this._addParts(args,t);
  865.   }
  866.   return args;
  867. };
  868.  
  869. FlashGotDMCust.prototype.createJob=function(links,opType) {
  870.   return { links: links, opType: opType };
  871. };
  872.  
  873. FlashGotDMCust.prototype.performJob=function(job) {
  874.   const links=job.links;
  875.   const exeFile=links.exeFile;
  876.   if(!exeFile) return;
  877.   
  878.   const len=links.length;
  879.   if(len < 1) return;
  880.   
  881.   const folder=links.folder;
  882.   const referrer = this.getReferrer(links);
  883.   const postData = links.postData;
  884.   var l, url;
  885.   
  886.   this.checkCookieSupport();
  887.   var cookieFile;
  888.   if(this.service.getPref("omitCookies")) {
  889.     cookieFile = null;
  890.   } else {
  891.     cookieFile = this.service.profDir.clone();
  892.     cookieFile.append("cookies.txt");
  893.     cookieFile = cookieFile.path;
  894.   }
  895.   
  896.   var j;
  897.   if(/\[[^\]]*UFILE[^\]]*\]/.test(this.argsTemplate)) {
  898.     // we must create a file list
  899.     var urlList = "";
  900.     for(j = 0; j < len; j++) {
  901.       urlList += links[j].href + "\n";
  902.     }
  903.     var args;
  904.     this.runNative(args = this.makeArgs(
  905.       null,
  906.       referrer,
  907.       null,
  908.       folder,
  909.       postData,
  910.       cookieFile,
  911.       this.createJobFile(urlList).path
  912.       ),
  913.     false, exeFile);
  914.     this.log("wget " + args.join(" "));
  915.   } else {
  916.    
  917.     for(j = 0; j < len; j++) {
  918.       l=links[j];
  919.       url=l.href;
  920.       this.runNative(  
  921.         this.makeArgs(
  922.           url, 
  923.           referrer, 
  924.           this.getCookie(l, links), 
  925.           folder, 
  926.           postData,
  927.           cookieFile,
  928.           null  // URL LIST FILE
  929.           ),
  930.       false, exeFile);
  931.       this.updateProgress(links, j, len);
  932.     }
  933.   }
  934. };
  935. FlashGotDMCust.prototype.checkExePlatform=function(exeFile) {
  936.   return exeFile;
  937. };
  938. FlashGotDMCust.prototype.createExecutable=function() {
  939.   return false;
  940. };
  941.  
  942.  
  943. // *****************************************************************************
  944. // END DMS CLASSES
  945. // *****************************************************************************
  946.  
  947. // DMS initialization
  948.  
  949. FlashGotDM.initDMS=function(service) {
  950.   var dm;
  951.   
  952.   dm=new FlashGotDM("Download Accelerator Plus");
  953.   dm.exclusive=true;
  954.   
  955.   new FlashGotDM("Download Master");
  956.   
  957.   new FlashGotDM("FlashGet");
  958.   
  959.   dm=new FlashGotDM("Free Download Manager");
  960.   dm._waitForNative=false;
  961.   
  962.   new FlashGotDM("FreshDownload");
  963.   
  964.   dm=new FlashGotDM("GetRight");
  965.   dm.super_download=FlashGotDM.prototype.download;
  966.   dm.super_createJob=FlashGotDM.prototype.createJob;
  967.   dm.download=function(links, opType) {
  968.     const service=this.service;
  969.     if(opType==service.OP_ONE && !service.getPref("GetRight.quick")) {
  970.       opType=service.OP_SEL;
  971.     }
  972.     this.super_download(links,opType);
  973.   };
  974.   dm.createJob=function(links, opType) {
  975.     const service=this.service;
  976.     var folder=links.folder;
  977.     if(!(folder && folder._fgSelected)) folder=false;
  978.     
  979.     var referrer=this.getReferrer(links);
  980.     
  981.     switch(opType) {
  982.       case service.OP_ONE:
  983.         var job=this.super_createJob(links,opType);
  984.         if(this.service.getPref("GetRight.old")) job+="\nold";
  985.         return job;
  986.       case service.OP_SEL:
  987.       case service.OP_ALL:
  988.         var urlList="";
  989.         var referrerLine=(referrer && referrer.length>0)?"\r\nReferer: "+referrer+"\r\n":"\r\n";
  990.         var l,decodedURL, urlParts, fileSpec, cookie;
  991.         for(var j=0, len=links.length; j<len; j++) {
  992.           l=links[j];
  993.           
  994.           if(folder) {
  995.             decodedURL=l.href;
  996.             try { decodedURL=decodeURI(decodedURL) } catch(ex) {};
  997.             urlParts=decodedURL.match(/\/\/.+[=\/]([^\/]+\.\w+)/);
  998.             if(!urlParts) urlParts=l.href.match(/.*\/(.*\w+.*)/);
  999.             if(urlParts && (fileSpec=urlParts[1])
  1000.               // && (links.length==1 ||  !/\.(php|[\w]?htm[l]?|asp|jsp|do|xml|rdf|\d+)$/i.test(fileSpec))
  1001.              ) {  
  1002.               urlParts=fileSpec.match(/(.*\.\w+).*/);
  1003.               if(urlParts) fileSpec=urlParts[1];
  1004.               fileSpec="File: "+folder+"\\"+fileSpec.replace(/[^\w\.-]/g,'_')+"\r\n";
  1005.             } else continue;
  1006.           } else fileSpec="";
  1007.           urlList+="URL: "+l.href
  1008.             +"\r\nDesc: "+l.description
  1009.             +referrerLine+fileSpec;
  1010.           if(cookie=this.getCookie(l,links)) {
  1011.             urlList+="Cookie: " + cookie + "\r\n";
  1012.           }
  1013.           this.updateProgress(links, j, len);
  1014.         }
  1015.         var file=service.tmpDir.clone();
  1016.         file.append("flashgot.grx");
  1017.         file.createUnique(0,0600);
  1018.         var charset=null;
  1019.         try {
  1020.           charset=service.getPref("GetRight.charset",
  1021.             service.prefService.QueryInterface(Components.interfaces.nsIPrefBranch
  1022.             ).getComplexValue("intl.charset.default",
  1023.               Components.interfaces.nsIPrefLocalizedString).data);
  1024.         } catch(ex) {}
  1025.         service.writeFile(file, urlList, charset);
  1026.         referrer=file.path;
  1027.         break;
  1028.     }
  1029.     var cmdOpts="/Q";
  1030.     if(service.getPref("GetRight.autostart",false)) { // CHECK ME!!!
  1031.       cmdOpts+="\n /AUTO";
  1032.     }
  1033.     return this.createJobHeader({ length: 0, folder: "" },opType) +
  1034.       referrer + "\n" + cmdOpts;
  1035.   };
  1036.   dm.askPath=[false,true,true];
  1037.   
  1038.   new FlashGotDM("HiDownload");
  1039.   new FlashGotDM("InstantGet");
  1040.   
  1041.   dm = new FlashGotDM("iGetter Win");
  1042.   dm.__defineGetter__("supported", 
  1043.     function() { return  "nsIGetterMoz" in Components.interfaces; });
  1044.   dm.createJob = function(links, opType) {
  1045.     var job = this.getReferrer(links) + "\r\n";
  1046.     for(var j=0; j < links.length; j++) {
  1047.       job += links[j].href + "\r\n" + links[j].description + "\r\n";
  1048.     }
  1049.     return job;
  1050.   };
  1051.   dm.performJob = function(job) {
  1052.     const file = this.createJobFile(job);
  1053.     delete job;
  1054.     Components.classes["@presenta/iGetter"]
  1055.               .getService(Components.interfaces.nsIGetterMoz)
  1056.               .NewURL(file.path);
  1057.     if(file.exists()) file.remove(0);
  1058.   };
  1059.   
  1060.   new FlashGotDM("Internet Download Accelerator");
  1061.   (new FlashGotDM("Internet Download Manager")).postSupport = true;
  1062.  
  1063.   var lg2002=new FlashGotDM("LeechGet 2002");
  1064.   var lg2004=new FlashGotDM("LeechGet");
  1065.   lg2004._bgJob=lg2002._bgJob=false;
  1066.   lg2004.super_createJob=lg2002.super_createJob=FlashGotDM.prototype.createJob;
  1067.   lg2004.createJob=lg2002.createJob=function(links, opType) {
  1068.     const service=this.service;
  1069.     var referrer;
  1070.     switch(opType) {
  1071.       case service.OP_ONE:
  1072.         return this.super_createJob(links, links.quickDownload?service.OP_ONE:service.OP_SEL);
  1073.       case service.OP_SEL:
  1074.         var htmlDoc="<html><head><title>FlashGot selection</title></head><body>";
  1075.         var l;
  1076.         for(var j=0, len=links.length; j<len; j++) {
  1077.           l=links[j];
  1078.           var des=l.description;
  1079.           var tag=l.tagName?l.tagName.toLowerCase():"";
  1080.           htmlDoc=htmlDoc.concat(tag=="img"
  1081.             ?"<img src=\""+l.href+"\" alt=\""+des
  1082.               +"\" width=\""+l.width+"\" height=\""+l.height+
  1083.               "\" />\n"
  1084.             :"<a href=\""+l.href+"\">"+des+"</a>\n");
  1085.           this.updateProgress(links,j,len);
  1086.         }
  1087.         referrer = service.httpServer.addDoc(
  1088.           htmlDoc.concat("</body></html>")
  1089.         );
  1090.         break;
  1091.        default:
  1092.         referrer=links.document.URL;
  1093.         if(referrer.match(/^\s*file:/i)) { // fix for local URLs
  1094.           // we serve local URLs through built-in HTTP server...
  1095.           return this.createJob(links,service.OP_SEL);
  1096.         }
  1097.     }
  1098.     return this.createJobHeader({ length: 0, folder: "" },opType)+referrer+"\n";
  1099.   };
  1100.  
  1101.   new FlashGotDM("Net Transport");
  1102.   new FlashGotDM("Net Transport 2");
  1103.   new FlashGotDM("NetAnts");
  1104.   new FlashGotDM("Mass Downloader");
  1105.   
  1106.   (new FlashGotDM("ReGet")).postSupport = true;
  1107.   
  1108.   const httpFtpValidator=function(url) {
  1109.     return /^(http:|ftp:)/.test(url);
  1110.   };
  1111.   dm=new FlashGotDM("Star Downloader");
  1112.   dm.cookieSupport=false;
  1113.   dm.isValidLink=httpFtpValidator;
  1114.   dm._waitForNative=false;
  1115.   
  1116.   dm=new FlashGotDM("TrueDownloader");
  1117.   dm.isValidLink=httpFtpValidator;
  1118.   dm._waitForNative=false;
  1119.   
  1120.   new FlashGotDM("Thunder");
  1121.   
  1122.   dm = new FlashGotDM("WellGet");
  1123.   dm.getRelativeExe = function() {
  1124.     try {
  1125.       return this.service.prefs.getComplexValue("WellGet.path", Components.interfaces.nsILocalFile);
  1126.     } catch(ex) {}
  1127.     return null;
  1128.   };
  1129.   dm.customSupportCheck = function() {
  1130.      try {
  1131.        var wellGetExe = this.getRelativeExe();
  1132.        var currentPath = wellGetExe.path;
  1133.        wellGetExe.initWithPath(this.service.profDir.path.substring(0,2) + dir.path.substring(2));
  1134.        if(wellGetExe.path != currentPath) {
  1135.           this.service.prefs.setComplexValue("WellGet.path", wellGetExe);
  1136.        }
  1137.        return wellGetExe.exists() && wellGetExe.isExecutable();
  1138.      } catch(ex) {
  1139.      }
  1140.      return false;
  1141.   };
  1142.   dm.createJob = function(links, opType) {
  1143.     var job = FlashGotDM.prototype.createJob.call(this, links, opType);
  1144.     var wellGetExe = this.getRelativeExe();
  1145.     if(wellGetExe) job += "\n" + wellGetExe.path;
  1146.     return job;
  1147.   };
  1148.   dm.shouldList = function() { return true; }
  1149.  
  1150.   dm=new FlashGotDMX("Aria","aria", "[-r REFERER] [-d FOLDER] -g [URL]");
  1151.   dm.createJob=function(links,opType) {
  1152.     return FlashGotDMX.prototype.createJob.call(this,links,opType) + "\nsleep 4\n" + this.unixCmd+" -s &\n";
  1153.   };
  1154.   dm._waitForNative=false;
  1155.   
  1156.   dm=new FlashGotDMX("Downloader 4 X (nt)","nt");
  1157.   dm.createJob=function(links,opType) {
  1158.     return this.unixCmd+"&\nsleep 1\n" +
  1159.       (links.folder && links.folder._fgSelected
  1160.       ? this.unixCmd+" -d '"+links.folder+"'\n"
  1161.       :"") + 
  1162.       FlashGotDMX.prototype.createJob.call(this,links,opType);
  1163.   };
  1164.   
  1165.   dm=new FlashGotDMX("Downloader 4 X","d4x","[--referer REFERER] [--directory FOLDER] [-a URL] [--al POST] [COOKIE]");
  1166.   dm.askPath=[false, true, true];
  1167.   dm.postSupport = true;
  1168.   dm.createJob = function(links, opType) {
  1169.     const service = this.service;
  1170.     const shellEsc = this.shellEsc;
  1171.     const referer = shellEsc(this.getReferrer(links));
  1172.     const folder = links.folder._fgSelected && links.folder || null;
  1173.     const quiet = service.getPref(this.codeName+".quiet",false);
  1174.     const len = links.length;
  1175.     var job;
  1176.     
  1177.     if(len>0) {
  1178.       
  1179.        var urls="";
  1180.        for(var j=0; j<len; j++) {
  1181.          urls+=" "+shellEsc(links[j].href);
  1182.          this.updateProgress(links, j, len);
  1183.        }
  1184.  
  1185.        var promptURLs_fakePost = null;
  1186.        var quietURLs_fakeCookie = null;
  1187.        
  1188.        if(quiet) {
  1189.          quietURLs_fakeCookie = urls;
  1190.          urls = null;
  1191.        } else if(len>1) {
  1192.          promptURLs_fakePost = urls;
  1193.          urls = null;
  1194.        }
  1195.        job = this.createCmdLine(
  1196.                   urls, 
  1197.                   referer,
  1198.                   quietURLs_fakeCookie || null,
  1199.                   folder,
  1200.                   promptURLs_fakePost);
  1201.     } else job = "";
  1202.     
  1203.     return job;
  1204.   };
  1205.   
  1206.   dm=new FlashGotDMX("KDE KGet","kget");
  1207.   dm.askPath=ASK_NEVER;
  1208.   
  1209.   dm=new FlashGotDMX("GNOME Gwget","gwget");
  1210.   dm.askPath=ASK_NEVER;
  1211.   
  1212.   dm=new FlashGotDMX("cURL","curl", '-L -O [--referer REFERER] [-b COOKIE] [-d POST] [URL]');
  1213.   dm.postSupport = true;
  1214.   dm.createJob=function(links,opType) {
  1215.     var job="[ -x \"`which 'xterm'`\" ] &&  CURL_CMD='xterm -e curl' || CURL_CMD='curl'\n";
  1216.     if (links.folder) job+="cd '"+links.folder+"'\n";
  1217.     this.unixCmd="$CURL_CMD";
  1218.     return job + FlashGotDMX.prototype.createJob.call(this,links,opType);
  1219.   };
  1220.  
  1221.   
  1222.   function FlashGotDMSD(version) {
  1223.     this._initMac("Speed Download "+version,"Spee");
  1224.     this.version=version;
  1225.   };
  1226.   FlashGotDMSD.prototype=new FlashGotDMMac();
  1227.   FlashGotDMSD.prototype.createJob=function(links,opType) {
  1228.     const v=this.version;
  1229.    
  1230.     var job = "tell app \""+ this.macAppName+ "\" to AddURL {";
  1231.     var urlList=[];
  1232.     var cookieList=[];
  1233.     var l;
  1234.     for(var j=0,len=links.length; j<len; j++) {
  1235.       l=links[j];
  1236.       urlList[urlList.length] = '"'+l.href+'"';
  1237.       if(v>2) {
  1238.         cookieList[cookieList.length] = '"'+this.getCookie(l,links)+'"';
  1239.       }
  1240.       this.updateProgress(links, j, len);
  1241.     }
  1242.     job+=urlList.join(',')+"}";
  1243.     if(v>2) {
  1244.       if(links.postData) { 
  1245.         job+=' with form data "'+links.postData+'"';
  1246.       }
  1247.       const referer=this.getReferrer(links);
  1248.       if(referer && referer.length) {
  1249.         job+=' from "'+referer+'"';
  1250.       }
  1251.       if(cookieList.length) {
  1252.         job+=' with cookies {' + cookieList.join(',') + '}';
  1253.       }
  1254.     }  
  1255.     return job;
  1256.   };
  1257.   
  1258.   new FlashGotDMSD(2);
  1259.   dm = new FlashGotDMSD(3);
  1260.   dm.cookieSupport = true;
  1261.   dm.postSupport = true;
  1262.   
  1263.   new FlashGotDMMac("iGetter","iGET");
  1264.   
  1265.   FlashGotDMCust.init(service);
  1266.   service.sortDMS();
  1267.   
  1268. };
  1269.  
  1270. // *****************************************************************************
  1271. // HTTP interceptor (nsIURIContentListener + http-on-modify-request observer)
  1272. // *****************************************************************************
  1273.  
  1274. function HttpInterceptor(service) {
  1275.   this.service = service;
  1276.  
  1277.   Components.classes["@mozilla.org/uriloader;1"].getService(
  1278.     Components.interfaces.nsIURILoader).registerContentListener(this);
  1279. }
  1280.  
  1281. HttpInterceptor.prototype = {
  1282.   service: null,
  1283.   
  1284.   autoStart: false,
  1285.   interceptAll: true,
  1286.   bypassAutoStart: false,
  1287.   forceAutoStart: false,
  1288.   
  1289.   lastPost: null, // last uploadChannel
  1290.   
  1291.   setup: function() { // profile initialization
  1292.     this.autoStart = this.service.getPref("autoStart", false);
  1293.     this.interceptAll = this.service.getPref("interceptAll", true);
  1294.   },
  1295.   
  1296.   dispose: function() {
  1297.     Components.classes["@mozilla.org/uriloader;1"].getService(
  1298.         Components.interfaces.nsIURILoader).unRegisterContentListener(this);
  1299.   },
  1300.   
  1301.   log: function(msg) {
  1302.     this.service.log(msg);
  1303.   },
  1304.   
  1305.   _shouldIntercept: function(contentType) {
  1306.     dump("FG: _shouldIntercept("+contentType+"\n");
  1307.     if(this.bypassAutoStart) return false;
  1308.     const service = this.service;
  1309.     if(!(service.DMS && service.DMS.found)) return false;
  1310.     if(this.forceAutoStart) return true;
  1311.     
  1312.     if(!this.autoStart) return false;
  1313.     
  1314.     if(this.interceptAll &&
  1315.       !/\bxpinstall\b/.test(contentType)) {
  1316.       return true;
  1317.     }
  1318.  
  1319.     if(contentType=="application/x-unknown-content-type") return false;
  1320.     var ms=Components.classes['@mozilla.org/uriloader/external-helper-app-service;1']
  1321.                      .getService(Components.interfaces.nsIMIMEService);
  1322.     const exts = service.extensions;
  1323.     for(var j = exts.length; j-->0;) {
  1324.       if(contentType == ms.getTypeFromExtension(exts[j])) return true;
  1325.     }
  1326.   }
  1327. , 
  1328.   _willHandle: function(url, contentType) {
  1329.     if(!/^(http|https|ftp|sftp|rtsp|mms):/i.test(url) ) {
  1330.       if((/^\s*javascript/i).test(url)) this.log("JavaScript url intercepted: "+url);
  1331.       return false;
  1332.     }
  1333.     return true;
  1334.   }
  1335. ,
  1336.   extractPostData: function(channel) {
  1337.     if(channel instanceof Components.interfaces.nsIUploadChannel &&
  1338.        channel.uploadStream instanceof Components.interfaces.nsISeekableStream) {
  1339.       this.log("Extracting post data...");
  1340.       try {
  1341.         channel.uploadStream.seek(0,0);
  1342.         const sis=Components.classes[
  1343.           '@mozilla.org/scriptableinputstream;1'].createInstance(
  1344.           Components.interfaces.nsIScriptableInputStream);
  1345.         sis.init(channel.uploadStream);
  1346.         var postData=sis.read(sis.available()).replace(/\s$/,'').split(/[\r\n]/);
  1347.         return postData[postData.length-1];
  1348.       } catch(ex) {
  1349.         this.log(ex.message);
  1350.       } finally {
  1351.          sis.close();
  1352.       }
  1353.     }
  1354.     return null;
  1355.   },
  1356.   /* nsIURIContentListener */
  1357.   
  1358.   canHandleContent: function(contentType, isContentPreferred, desiredContentType) {
  1359.     dump("FG: canHandleContent "+contentType+"\n");
  1360.     return this._shouldIntercept(contentType);
  1361.   }
  1362. ,
  1363.   doContent: function(contentType, isContentPreferred, channel, contentHandler) {
  1364.     const ci=Components.interfaces;
  1365.     channel.QueryInterface(ci.nsIChannel);
  1366.     const uri=channel.URI;
  1367.     dump("FG: doContent "+contentType+" "+uri.spec+"\n");
  1368.     if(!this._willHandle(uri.spec,contentType)) {
  1369.       throw new Error("FlashGot not interested in "+contentType+" from "+uri.spec);
  1370.     }
  1371.     
  1372.     this.log("Intercepting download...");
  1373.  
  1374.     const pathParts=uri.path.split(/\//);
  1375.     var links=[ {
  1376.      href: channel.URI.spec, 
  1377.      title: pathParts[pathParts.length-1],
  1378.     } ];
  1379.     
  1380.     if(channel instanceof ci.nsIHttpChannel) {
  1381.       links.referrer=channel.referrer.spec;
  1382.       links.postData = this.extractPostData(channel);
  1383.     }
  1384.       
  1385.     if(this.service.download(links)) {
  1386.       this.log("...interception done!");
  1387.     } else {
  1388.        throw new Error("Can't download from this URL: "+uri.spec);
  1389.     }
  1390.     
  1391.     channel.cancel(Components.results.NS_BINDING_ABORTED); 
  1392.    
  1393.     this.log("Original request cancelled.");
  1394.     contentHandler.value=null;
  1395.     return true;
  1396.   }
  1397. ,
  1398.   isPreferred: function(contentType, desiredContentType) {
  1399.     dump("FG: isPreferred("+contentType+","+desiredContentType+"\n");
  1400.     return this._shouldIntercept(contentType);
  1401.   }
  1402. ,
  1403.   onStartURIOpen: function(uri) {
  1404.     dump("FG: onStartURIOpen "+ uri + (uri && uri.spec) + "\n");
  1405.     return false;
  1406.   }
  1407. ,
  1408.   /* http-on-modify-request Observer */
  1409.   observe: function(channel, topic, data) {
  1410.     if(channel instanceof Components.interfaces.nsIUploadChannel) {
  1411.       this.lastPost = channel.QueryInterface(Components.interfaces.nsIHttpChannel);
  1412.     }
  1413.   }
  1414. }
  1415.  
  1416.  
  1417.  
  1418.  
  1419. // *****************************************************************************
  1420. // XPCOM Service
  1421. // *****************************************************************************
  1422.  
  1423. const SHUTDOWN="profile-before-change";
  1424. const STARTUP="profile-after-change";
  1425.  
  1426. function FlashGotService() {
  1427.   
  1428.   const osvr=Components.classes['@mozilla.org/observer-service;1'].getService(
  1429.     Components.interfaces.nsIObserverService);
  1430.   
  1431.   osvr.addObserver(this,SHUTDOWN,false);
  1432.   osvr.addObserver(this,"xpcom-shutdown",false);
  1433.   osvr.addObserver(this,STARTUP,false);
  1434.   
  1435.   this.interceptor = new HttpInterceptor(this);
  1436.   osvr.addObserver(this.interceptor, "http-on-modify-request", false);
  1437.   
  1438.   
  1439. }
  1440.  
  1441. FlashGotService.prototype = {
  1442.   OP_ONE: 0, 
  1443.   OP_SEL: 1,
  1444.   OP_ALL: 2,
  1445.   OP_QET: 3
  1446. ,
  1447.   get wrappedJSObject() {
  1448.     return this;
  1449.   }
  1450. ,
  1451.   unregister: function() {
  1452.     try {
  1453.       const osvr=Components.classes['@mozilla.org/observer-service;1'].getService(
  1454.       Components.interfaces.nsIObserverService);
  1455.       osvr.removeObserver(this,SHUTDOWN);
  1456.       osvr.removeObserver(this,"xpcom-shutdown");
  1457.       osvr.removeObserver(this,STARTUP);
  1458.       osvr.removeObserver(this.interceptor, "http-on-modify-request");
  1459.       this.interceptor.dispose();
  1460.     } catch(ex) {
  1461.       this.log("Error unregistering service as observer: "+ex);
  1462.     }
  1463.   }
  1464. ,
  1465.   QueryInterface: function(iid) {
  1466.      xpcom_checkInterfaces(iid,SERVICE_IIDS,Components.results.NS_ERROR_NO_INTERFACE);
  1467.      return this;
  1468.   }
  1469. ,
  1470.   /* nsIObserver */  
  1471.   observe: function(subject, topic, data) {
  1472.     if(subject == this.prefs) {
  1473.       this.syncPrefs(data);
  1474.     } else {
  1475.       switch(topic) {
  1476.         case "xpcom-shutdown":
  1477.           this.unregister();
  1478.           break;
  1479.         case SHUTDOWN: 
  1480.           this.cleanup();
  1481.           break;
  1482.         case STARTUP:
  1483.           this.initGlobals();
  1484.           this.interceptor.setup();
  1485.           break;
  1486.       }
  1487.     }
  1488.   }
  1489. ,
  1490.   syncPrefs: function(name) {
  1491.     this.logEnabled=this.getPref("logEnabled",true);
  1492.     if(name) {
  1493.       switch(name) {
  1494.         case "hide-icons":
  1495.           var w;
  1496.           for(var wins=this.windowMediator.getEnumerator(null); wins.hasMoreElements();) {
  1497.              w=wins.getNext();
  1498.              if(typeof(w.gFlashGot)=="object" && w.gFlashGot.toggleMainMenuIcon) {
  1499.                w.gFlashGot.toggleMainMenuIcon();
  1500.              }
  1501.           }
  1502.         break;
  1503.         
  1504.         case "autoStart":
  1505.         case "interceptAll":
  1506.           this.interceptor[name] = this.getPref("name");
  1507.         break;
  1508.       }
  1509.     }
  1510.   }
  1511. ,
  1512.   
  1513.   get defaultDM() {
  1514.     return this.getPref("defaultDM",null);
  1515.   }
  1516. ,
  1517.   set defaultDM(name) {
  1518.     this.setPref("defaultDM", name);
  1519.     return name;
  1520.   }
  1521. ,
  1522.   get tmpDir() {
  1523.     return this.globals.tmpDir; 
  1524.   }
  1525. ,
  1526.   get profDir() {
  1527.     return this.globals.profDir; 
  1528.   }
  1529. ,
  1530.   get DMS() {
  1531.     return this.globals.DMS;
  1532.   }
  1533. ,
  1534.   get extensions() {
  1535.     var s=this.getPref("extensions","");
  1536.     return s?s.split(','):[];
  1537.   }
  1538. ,
  1539.   set extensions(v) {
  1540.     var arr=null;
  1541.     var s = typeof(v)=="string" 
  1542.       ? v : typeof(v)=="object" && typeof(v.join)=="function" 
  1543.         ? (arr=v).join(',').replace(/[^\w\-,]/g,"") : "";
  1544.     this.setPref("extensions", s);
  1545.     return arr?arr:[];
  1546.   }
  1547. ,
  1548.   addExtension: function(ext) {
  1549.     if(ext) {
  1550.       var extensions=this.extensions;
  1551.       if(!this.extensionExists(ext,extensions)) {
  1552.         extensions[extensions.length] = ext;
  1553.         extensions.sort();
  1554.         this.extensions=extensions;
  1555.         return true;
  1556.       }
  1557.     }
  1558.     return false;
  1559.   }
  1560. ,
  1561.   removeExtension: function(ext) {
  1562.     var extensions=this.extensions;
  1563.     var j=this.indexOfExtension(ext,extensions);
  1564.     if(j>-1) {
  1565.       extensions.splice(j,1);
  1566.       this.extensions=extensions;
  1567.       return true;
  1568.     }
  1569.     return false;
  1570.   }
  1571. ,
  1572.   extensionExists: function(ext,extensions) {
  1573.     return this.indexOfExtension(ext,extensions)>-1;
  1574.   }
  1575. ,
  1576.   indexOfExtension: function(ext,extensions) {
  1577.     var ext=ext.toLowerCase();
  1578.     if(typeof(extensions)!="object") extensions=this.extensions;
  1579.     for(var j=extensions.length; j-->0;) {
  1580.       if(extensions[j].toLowerCase()==ext) return j;
  1581.     }
  1582.     return -1;
  1583.   }
  1584. ,
  1585.   _httpServer: null,
  1586.   get httpServer() {
  1587.     if(typeof(FlashGotHttpServer) != "function") {
  1588.       Components.classes["@mozilla.org/moz/jssubscript-loader;1"]
  1589.           .getService(Components.interfaces.mozIJSSubScriptLoader)
  1590.           .loadSubScript("chrome://flashgot/content/flashgotHttpServer.js", null);
  1591.     }
  1592.     return ((!this._httpServer) || this._httpServer.isDown) ?
  1593.        this._httpServer=new FlashGotHttpServer(this)
  1594.       :this._httpServer;
  1595.   }
  1596.  
  1597. ,
  1598.   download: function(links, opType, dmName) {
  1599.     if(links.length == 0) return false;
  1600.     if(!opType) opType=links.length>1?this.OP_SEL:this.OP_ONE;
  1601.     if(!dmName) dmName=this.defaultDM;
  1602.     const dm=this.DMS[dmName];
  1603.     if(!dm) {
  1604.       this.log("FlashGot error: no download manager selected!");
  1605.       return false;
  1606.     }
  1607.     
  1608.     // surrogate missing attributes
  1609.     
  1610.     if(!links.progress) {
  1611.       links.progress = { update: function() {} };
  1612.     } else {
  1613.       links.progress.update(12);
  1614.     }
  1615.     
  1616.    
  1617.     var service = this;
  1618.     this._delay(function(t) { service._downloadDelayed(links, opType, dm); }); 
  1619.     return true;
  1620.   },
  1621.   
  1622.   _downloadDelayed: function(links, opType, dm) {
  1623.     
  1624.      if(!links.postData) { 
  1625.       links.postData = "";
  1626.     } else if(!dm.postSupported) {
  1627.       // surrogate POST parameters as query string
  1628.       links[0].href += (links[0].href.indexOf("?") > -1 ?  "&" : "?") + links.postData;
  1629.     }
  1630.  
  1631.     const encodedURLs=this.getPref(dm.codeName+".encode",this.getPref("encode",true));
  1632.  
  1633.     const extFilter = this.getPref("extfilter", false) && !this.interceptor.interceptAll ?
  1634.         new RegExp("\.(" +
  1635.           this.extensions.join("|").replace(/[^\w-|]/,"") + 
  1636.           ")\\b", "i") : null;
  1637.     
  1638.     var logMsg = "Processing "+links.length+" links ";
  1639.     if(this.logEnabled && typeof(links.startTime) == "number") {
  1640.       logMsg += "scanned in ms" + (Date.now() - links.startTime);
  1641.     }
  1642.     
  1643.     
  1644.  
  1645.     var startTime = Date.now();
  1646.     const pg=links.progress;
  1647.     
  1648.     const escapeCheckNo=/(%[0-9a-f]{2,4})/i;
  1649.     const escapeCheckYes=/[\s]+/;
  1650.     
  1651.     var len = links.length;
  1652.     
  1653.     var filters = null;
  1654.     if(len > 1) {
  1655.       filters = [];
  1656.       
  1657.       const isValid = dm.isValidLink; 
  1658.       if(isValid)  filters.push(function() { return isValid(href) });
  1659.       
  1660.       if(extFilter) filters.push(function() { return extFilter.test(href) });
  1661.       
  1662.       if(filters.length) {
  1663.         filters.doFilter = function(href) {
  1664.           for(var j = this.length; j-- > 0;) if(!this[j](href)) return false;
  1665.           return true;
  1666.         }
  1667.       } else {
  1668.         filters = null;
  1669.       }
  1670.     }
  1671.  
  1672.     const map = {};
  1673.     
  1674.     var k, j, l, href, ol, pos1, pos2;
  1675.     var jCount;
  1676.     // divide et impera :)
  1677.     var chunkLen = 100;
  1678.     const chunks = Math.ceil(len / chunkLen); 
  1679.     j = 0;
  1680.     for(k = 1; k <= chunks; k++) {
  1681.       if(k == chunks) chunkLen = len % chunkLen;
  1682.       if(j > 0) {
  1683.         pg.update(10 + 30 * j / len);
  1684.       }
  1685.      
  1686.       for(jCount = j + chunkLen; j < jCount; j++) {
  1687.         l = links[j];
  1688.         l._pos = j;
  1689.         href = l.href;
  1690.         if((!filters) || filters.doFilter(href)) {
  1691.           ol = map[href];
  1692.           if(ol) { // duplicate, keep the longest description
  1693.             if(ol.description.length < l.description.length) {
  1694.               map[href]=l;
  1695.               l.href = ol.href; // keep sanitizations
  1696.             }
  1697.           } else {
  1698.             map[href] = l;
  1699.             // encoding checks
  1700.             try {
  1701.               if(encodedURLs) { 
  1702.                 if(escapeCheckYes.test(href) || !escapeCheckNo.test(href)) { 
  1703.                   href=encodeURI(href);
  1704.                 }
  1705.                 // workaround for malformed hash urls... I have still to think about it :-)
  1706.                 while((pos1=href.indexOf('#'))>-1 
  1707.                   && ((pos2=href.indexOf('?'))<0 
  1708.                       || pos2>pos1 || pos1!=href.lastIndexOf('#'))) {
  1709.                   href=href.substring(0,pos1)+'%23'+href.substring(pos1+1);
  1710.                 }
  1711.                 l.href=href;
  1712.               } else {  
  1713.                 l.href=decodeURI(href);
  1714.               }
  1715.             } catch(e) {
  1716.               dump("Problem "
  1717.                 + ( encodedURLs ? "escaping" : "unescaping")
  1718.                 + " URL " + href + ": "+ e.message + "\n");
  1719.             }
  1720.           }
  1721.         }
  1722.       }
  1723.     }
  1724.    
  1725.     links.length = 0;
  1726.     for(href in map) links[links.length] = map[href];
  1727.     
  1728.     if(links.length>1) {
  1729.       dm.log("Sorting again "+links.length+" links");
  1730.       links.sort(function(a,b) {
  1731.         a=a._pos; b=b._pos;
  1732.         return a>b?1:a<b?-1:0;
  1733.       });
  1734.     }
  1735.     pg.update(50);
  1736.     
  1737.     dm.log("Preprocessing done in ms" + (Date.now() - startTime) );
  1738.     
  1739.     // "true" download
  1740.     this._delay(function(t) {
  1741.         dm.log("Starting dispatch");
  1742.         var startTime = Date.now();
  1743.     
  1744.         dm.download(links, opType);
  1745.  
  1746.         var now = Date.now();
  1747.         var logMsg = "Dispatch done in ms" + (now - startTime);
  1748.         if(typeof(links.startTime) == "number") { 
  1749.           logMsg += "\nTotal processing time: ms" + (now - links.startTime);
  1750.         }  
  1751.         dm.log(logMsg);
  1752.       });
  1753.   }
  1754. ,
  1755.   _delay: function(callback, time) {
  1756.      var timerCallback = { notify: callback }; 
  1757.      Components.classes["@mozilla.org/timer;1"].createInstance(Components.interfaces.nsITimer)
  1758.               .initWithCallback(timerCallback, time || 0, Components.interfaces.nsITimer.TYPE_ONE_SHOT);
  1759.   }
  1760. ,
  1761.   yeld: function() {
  1762.     const eqs = Components.interfaces.nsIEventQueueService;
  1763.     Components.classes["@mozilla.org/event-queue-service;1"]
  1764.       .getService(eqs).getSpecialEventQueue(eqs.UI_THREAD_EVENT_QUEUE)
  1765.       .processPendingEvents();
  1766.   },
  1767.   
  1768.   get bgProcessing() {
  1769.     return false;
  1770.       // this.getPref("bgProcessing", true);
  1771.   }
  1772. ,
  1773.   get prefService() {
  1774.     return Components.classes["@mozilla.org/preferences-service;1"].getService(
  1775.       Components.interfaces.nsIPrefService);
  1776.   }
  1777. ,
  1778.   savePrefs: function() {
  1779.     return this.prefService.savePrefFile(null);
  1780.   }
  1781. ,
  1782.   getPref: function(name,def) {
  1783.     const IPC=Components.interfaces.nsIPrefBranch;
  1784.     const prefs=this.prefs;
  1785.     try {
  1786.       switch(prefs.getPrefType(name)) {
  1787.         case IPC.PREF_STRING:
  1788.           return prefs.getCharPref(name);
  1789.         case IPC.PREF_INT:
  1790.           return prefs.getIntPref(name);
  1791.         case IPC.PREF_BOOL:
  1792.           return prefs.getBoolPref(name);
  1793.       }
  1794.     } catch(e) {}
  1795.     return def;
  1796.   }
  1797. ,
  1798.   setPref: function(name,value) {
  1799.     const prefs=this.prefs;
  1800.     switch(typeof(value)) {
  1801.       case "string":
  1802.           prefs.setCharPref(name,value);
  1803.           break;
  1804.       case "boolean":
  1805.         prefs.setBoolPref(name,value);
  1806.         break;
  1807.       case "number":
  1808.         prefs.setIntPref(name,value);
  1809.         break;
  1810.       default:
  1811.         throw new Error("Unsupported type "+typeof(value)+" for preference "+name);
  1812.     }
  1813.   }
  1814. ,
  1815.   _bundle: null,
  1816.   get bundle() {
  1817.     if(!this._bundle) {
  1818.       function getBundle(url) {
  1819.         try {
  1820.           var bundle = bs.createBundle(url);
  1821.           bundle.GetStringFromName("flashgot");
  1822.         } catch(ex) {
  1823.           dump("\n"+ex+"\n");
  1824.           bundle = null;
  1825.         }
  1826.         return bundle;
  1827.       }
  1828.       
  1829.       var bs = Components.classes["@mozilla.org/intl/stringbundle;1"].getService(
  1830.         Components.interfaces.nsIStringBundleService);
  1831.       if(! ( 
  1832.             (this._bundle=getBundle("chrome://flashgot/locale/flashgot.properties") )
  1833.             || (this._bundle=bs.createBundle("chrome://flashgot/content/en-US/flashgot.properties"))
  1834.             )
  1835.         ) {
  1836.         this._bundle = {
  1837.           formatStringFromName: function(name,parms) {
  1838.             return name+" ["+parms.join(',')+"]"
  1839.           },
  1840.           GetStringFromName: function(name) {
  1841.             return name;
  1842.           }
  1843.         };
  1844.       }
  1845.     }
  1846.     return this._bundle;
  1847.   }
  1848. ,
  1849.   getString: function(name,parms) {
  1850.     const bundle=this.bundle;
  1851.     try {
  1852.       return (parms
  1853.           ?bundle.formatStringFromName(name,parms,parms.length)
  1854.           :bundle.GetStringFromName(name));
  1855.     } catch(ex) {
  1856.       return "???";
  1857.     }
  1858.   }
  1859. ,
  1860.   _logFile: null,
  1861.   get logFile() {
  1862.     if(this._logFile==null) {
  1863.       this._logFile=this.profDir.clone();
  1864.       this._logFile.append("flashgot.log");
  1865.     }
  1866.     return this._logFile;
  1867.   }
  1868. ,
  1869.   logStream: null,
  1870.   logEnabled: false,
  1871.   log: function(msg) {
  1872.     if(this.logEnabled) {
  1873.       try {
  1874.         if(!this.logStream) {
  1875.           const logFile=this.logFile;
  1876.           const logStream=Components.classes["@mozilla.org/network/file-output-stream;1"
  1877.             ].createInstance(Components.interfaces.nsIFileOutputStream );
  1878.           logStream.init(logFile, 0x02 | 0x08 | 0x10, 0600, 0 );
  1879.           this.logStream=logStream;
  1880.           const header="*** Log start at "+new Date().toGMTString()+"\n";
  1881.           this.logStream.write(header,header.length);
  1882.         }
  1883.         
  1884.         if(msg!=null) {
  1885.           msg+="\n";
  1886.           this.logStream.write(msg,msg.length);
  1887.         }
  1888.         this.logStream.flush();
  1889.       } catch(ex) {
  1890.         dump(ex.message+"\noccurred logging this message:\n"+msg);
  1891.       }
  1892.     }
  1893.   }
  1894. ,
  1895.   dumpStack: function(msg) {
  1896.     dump( (msg?msg:"")+"\n"+new Error().stack+"\n");
  1897.   }
  1898. ,
  1899.   clearLog: function() {
  1900.     try {
  1901.       if(this.logStream) {
  1902.         try {
  1903.           this.logStream.close();
  1904.         } catch(eexx) {
  1905.           dump(eexx.message);
  1906.         }
  1907.       }
  1908.       if(this.logFile) this.logFile.remove(true);
  1909.       this.logStream=null;
  1910.       this.log(null);
  1911.     } catch(ex) { dump(ex.message); }
  1912.   } 
  1913. ,
  1914.   get windowMediator() {
  1915.     return Components.classes["@mozilla.org/appshell/window-mediator;1"
  1916.       ].getService(Components.interfaces.nsIWindowMediator);
  1917.   }
  1918. ,
  1919.   getWindow: function() {
  1920.     return this.windowMediator.getMostRecentWindow(null);
  1921.   }
  1922. ,
  1923.   _globals: null,
  1924.   get globals() {
  1925.     if(!this._initialized) {
  1926.       this.initGlobals();
  1927.     }
  1928.     return this._globals;
  1929.   }
  1930. ,
  1931.   PREFS_BRANCH: "flashgot."
  1932. ,
  1933.   _prefs: null,
  1934.   get prefs() {
  1935.     var prefs=this._prefs;
  1936.     if(!prefs) {
  1937.       this._prefs=prefs=this.prefService.getBranch(this.PREFS_BRANCH
  1938.         ).QueryInterface(Components.interfaces.nsIPrefBranchInternal);
  1939.     }
  1940.     return prefs;
  1941.   }
  1942. ,
  1943.   _initialized: false,
  1944.   initGlobals: function() {
  1945.     if(this._globals || this._initialized) return;
  1946.     
  1947.     function prepareTmp(t) {
  1948.       t.append("flashgot."+encodeURI(profDir.leafName).replace(/%/g,"_"));
  1949.       if(t.exists()) {
  1950.        if(!t.isDirectory()) t.createUnique(1,0700);
  1951.       } else {
  1952.         t.create(1,0700);
  1953.       }
  1954.       return t;
  1955.     }
  1956.     
  1957.     try {
  1958.       const startTime = Date.now();
  1959.       const prefs=this.prefs;
  1960.       const cc=Components.classes;
  1961.       const ci=Components.interfaces; 
  1962.  
  1963.       const fileLocator=cc["@mozilla.org/file/directory_service;1"].getService(
  1964.         ci.nsIProperties);
  1965.       const profDir=fileLocator.get("ProfD",ci.nsIFile);
  1966.      
  1967.       var tmpDir;
  1968.       try {
  1969.         tmpDir=prepareTmp(prefs.getComplexValue("tmpDir", ci.nsILocalFile));
  1970.       } catch(ex) {
  1971.         tmpDir=prepareTmp(fileLocator.get("TmpD", ci.nsILocalFile));
  1972.       }
  1973.        
  1974.       this._globals={
  1975.         tmpDir: tmpDir,
  1976.         profDir: profDir,
  1977.         prefs: prefs
  1978.       };
  1979.       
  1980.       prefs.addObserver("", this, false);
  1981.       this.syncPrefs();
  1982.       
  1983.       this.log("Per-session init started");
  1984.         
  1985.       this._setupLegacyPrefs();
  1986.  
  1987.       this._globals.DMS=this.checkDownloadManagers(true, false);
  1988.       this.log("Per-session init done in " + (Date.now() - startTime) + "ms");
  1989.     } catch(initEx) {
  1990.       this._initException=initEx;
  1991.     }
  1992.     this._initialized=true; 
  1993.   }
  1994. ,
  1995.   dispose: function() {
  1996.     this.prefs.removeObserver("",this);
  1997.     this._prefs=null;
  1998.     this._initialized=false;
  1999.     this._globals=null;
  2000.   }
  2001. ,
  2002.   createCustomDM: function(name) {
  2003.     const dm=new FlashGotDMCust(name);
  2004.     if(name && name.length) {
  2005.       FlashGotDMCust.persist(this);
  2006.       this.sortDMS();
  2007.       this.checkDownloadManagers(false,false);
  2008.     }
  2009.     return dm;
  2010.   }
  2011. ,
  2012.  removeCustomDM: function(name) {
  2013.    const dms=FlashGotDM.dms;
  2014.    for(var j=dms.length; j-->0;) {
  2015.      if(dms[j].custom && dms[j].name==name) {
  2016.        dms.splice(j,1);
  2017.        delete dms[name];
  2018.      }
  2019.    }
  2020.    FlashGotDMCust.persist(this);
  2021.    this.checkDownloadManagers(false,false);
  2022.  }
  2023. ,
  2024.   sortDMS: function() {
  2025.     FlashGotDM.dms.sort(function(a,b) { a=a.name.toLowerCase(); b=b.name.toLowerCase(); return a>b?1:a<b?-1:0; });
  2026.   }
  2027. , 
  2028.   checkDownloadManagers: function(init,detect) {
  2029.     
  2030.     if(init || detect) {
  2031.       FlashGotDM.init(this);
  2032.     }
  2033.     
  2034.     const dms=FlashGotDM.dms;
  2035.     dms.found=false;
  2036.     var defaultDM=this.defaultDM;
  2037.     if(!dms[defaultDM]) defaultDM=null;
  2038.     
  2039.     detect=detect || this.getPref("detect.auto",true);
  2040.  
  2041.     var j,dm;
  2042.     var cache;
  2043.     
  2044.     if(!detect) {
  2045.       cache=this.getPref("detect.cache","").split(",");
  2046.       for(j=dms.length; j-->0;) {
  2047.         dm=dms[j]._supported = false;
  2048.       }
  2049.       var name;
  2050.       for(j=cache.length; j-->0;) {
  2051.         name=cache[j];
  2052.         if(name.length && typeof(dm=dms[name])=="object" && dm.name==name) {
  2053.           dm._supported = true;
  2054.         }
  2055.       }
  2056.     }
  2057.     
  2058.     cache=[];
  2059.     var exclusive;
  2060.     var firstSupported=null;
  2061.     for(j=dms.length; j-- >0;) {
  2062.       dm=dms[j];
  2063.       if(dm.supported) {
  2064.         dms.found=true;
  2065.         cache[cache.length] = firstSupported = dm.name;
  2066.         if(dm.exclusive) exclusive=true;
  2067.       } else {
  2068.         this.log("Warning: download manager "+dm.name+" not found");
  2069.         if(defaultDM==dm.name) {
  2070.           defaultDM=null;
  2071.           this.log(dm.name+" was default download manager: resetting.");
  2072.         }
  2073.       }
  2074.     }
  2075.     
  2076.     this.setPref("detect.cache",cache.join(","));
  2077.     
  2078.     if( (!defaultDM) && firstSupported!=null) {
  2079.       this.defaultDM=firstSupported;
  2080.       this.log("Default download manager set to "+this.defaultDM);
  2081.     } else if(!dms.found) {
  2082.       this.log("Serious warning! no supported download manager found...");
  2083.     } 
  2084.     if(exclusive) {
  2085.       for(j=dms.length; j-->0;) {
  2086.         if(! (dms[j].custom || dms[j].exclusive) ) {
  2087.           dms.splice(j,1);
  2088.         }
  2089.       }
  2090.     }
  2091.     
  2092.     return dms;
  2093.   }
  2094. ,
  2095.   _referrerSpoofer: null,
  2096.   get referrerSpoofer() {
  2097.     if(typeof(ReferrerSpoofer) != "function") {
  2098.       Components.classes["@mozilla.org/moz/jssubscript-loader;1"]
  2099.           .getService(Components.interfaces.mozIJSSubScriptLoader)
  2100.           .loadSubScript("chrome://flashgot/content/referrerSpoofer.js", null);
  2101.     }
  2102.     return (!this._httpServer) ? this._referrerSpoofer = new ReferrerSpoofer() :this._referrerSpoofer;
  2103.   }
  2104. ,
  2105.   _cleaningup: false
  2106. ,
  2107.   cleanup: function() {
  2108.     if(this._cleaningup) return;
  2109.     try {
  2110.       this._cleaningup=true;
  2111.       this.log("Starting cleanup");
  2112.       if(this._httpServer) {
  2113.         this._httpServer.shutdown();
  2114.       }
  2115.       
  2116.       try {
  2117.         FlashGotDM.cleanup();
  2118.       } catch(eexx) {
  2119.         dump(eexx.message);
  2120.       }
  2121.       
  2122.       if(this._globals && this._globals.tmpDir.exists()) {
  2123.         try {
  2124.           this._globals.tmpDir.remove(true);
  2125.         } catch(eexx) {
  2126.           this.log("Can't remove "+this._globals.tmpDir.path+", maybe still in use: "+eexx);
  2127.         }
  2128.       }
  2129.       this._bundle=null;
  2130.       this.log("Cleanup done");
  2131.       if(this._logFile) try {
  2132.         if(this.logStream) this.logStream.close();
  2133.         var maxLogSize=Math.max(Math.min(this.getPref('maxLogSize',100000),1000000),50000);
  2134.         const logFile=this.logFile;
  2135.         const logSize=logFile.fileSize;
  2136.         if(logSize>maxLogSize) { // log rotation
  2137.           dump("Cutting log (size: "+logSize+", max: "+maxLogSize+")");
  2138.           const cc=Components.classes;
  2139.           const ci=Components.interfaces;
  2140.          
  2141.           const logBak=logFile.clone();
  2142.           logBak.leafName=logBak.leafName+".bak";
  2143.           if(logBak.exists()) logBak.remove(true);
  2144.           logFile.copyTo(logBak.parent,logBak.leafName);
  2145.           const is=cc['@mozilla.org/network/file-input-stream;1'].createInstance(
  2146.             ci.nsIFileInputStream);
  2147.           is.init(logBak,0x01, 0400, null);
  2148.           is.QueryInterface(ci.nsISeekableStream);
  2149.           is.seek(ci.nsISeekableStream.NS_SEEK_END,-maxLogSize);
  2150.           const sis=cc['@mozilla.org/scriptableinputstream;1'].createInstance(
  2151.           ci.nsIScriptableInputStream);
  2152.           sis.init(is);
  2153.           var buffer;
  2154.           var content="\n";
  2155.           var logStart=-1;
  2156.           while(buffer=sis.read(5000)) {
  2157.             content+=buffer;
  2158.             if((logStart=content.indexOf("\n*** Log start at "))>-1) { 
  2159.               content=content.substring(logStart);
  2160.               break;
  2161.             }
  2162.             content=buffer;
  2163.           }
  2164.           if(logStart>-1) {
  2165.              const os=cc["@mozilla.org/network/file-output-stream;1"].createInstance(
  2166.               ci.nsIFileOutputStream);
  2167.             os.init(logFile,0x02 | 0x08 | 0x20, 0700, 0);
  2168.             os.write(content,content.length);
  2169.             while(buffer=sis.read(20000)) {
  2170.               os.write(buffer,buffer.length);
  2171.             } 
  2172.             os.close();
  2173.           }
  2174.           sis.close();
  2175.         }
  2176.       } catch(eexx) {
  2177.         dump("Error cleaning up log: "+eexx);
  2178.       }
  2179.       this.logStream=null;
  2180.     } catch(ex) {
  2181.        this.log(ex);
  2182.     }
  2183.     this._cleaningup=false;
  2184.     this.dispose();
  2185.   }
  2186. ,
  2187.   readFile: function(file) {
  2188.     const cc=Components.classes;
  2189.     const ci=Components.interfaces; 
  2190.     
  2191.     const is = cc["@mozilla.org/network/file-input-stream;1"].createInstance(
  2192.           ci.nsIFileInputStream );
  2193.     is.init(file ,0x01, 0400, null);
  2194.     const sis = cc["@mozilla.org/scriptableinputstream;1"].createInstance(
  2195.       ci.nsIScriptableInputStream );
  2196.     sis.init(is);
  2197.     const res=sis.read(sis.available());
  2198.     is.close();
  2199.     return res;
  2200.   }
  2201. ,
  2202.   writeFile: function(file, content, charset) {
  2203.     const cc=Components.classes;
  2204.     const ci=Components.interfaces;
  2205.     const unicodeConverter = cc["@mozilla.org/intl/scriptableunicodeconverter"].createInstance(
  2206.     ci.nsIScriptableUnicodeConverter);
  2207.     try {
  2208.       unicodeConverter.charset = charset?charset:"UTF-8";
  2209.     } catch(ex) {
  2210.       unicodeConverter.charset = "UTF-8";
  2211.     }
  2212.     content=unicodeConverter.ConvertFromUnicode(content);
  2213.     const os=cc["@mozilla.org/network/file-output-stream;1"].createInstance(
  2214.       ci.nsIFileOutputStream);
  2215.     os.init(file,0x02,0700,0);
  2216.     os.write(content,content.length);
  2217.     os.close();
  2218.   }
  2219. ,
  2220.   _lookupMethod: null,
  2221.   get lookupMethod() {
  2222.     return this._lookupMethod?this._lookupMethod:(this._lookupMethod = 
  2223.       (Components.utils && Components.utils.lookupMethod)
  2224.         ?Components.utils.lookupMethod:Components.lookupMethod);
  2225.   }
  2226. ,
  2227.   _setupLegacyPrefs: function() {
  2228.     try {
  2229.       const file=this._globals.profDir.clone();
  2230.       const defFile=file.clone();
  2231.       file.append("pref");
  2232.       file.append("flashgot.js");
  2233.       defFile.append("prefs.js");
  2234.       if(file.exists() && defFile.exists()) {
  2235.         this.prefService.readUserPrefs(file);
  2236.         this.prefService.readUserPrefs(defFile);
  2237.         this.savePrefs();
  2238.         file.remove(true);
  2239.       }
  2240.     } catch(e) {
  2241.       this.log(e.message);
  2242.     }
  2243.   }
  2244. ,
  2245.   showDMSReference: function() {
  2246.     this.getWindow().open("http://www.flashgot.net/dms","_blank");
  2247.   }
  2248. , 
  2249.   dirtyJobsDone: false
  2250. }
  2251.  
  2252. // XPCOM Scaffolding code
  2253.  
  2254. // component defined in this file
  2255.  
  2256. const SERVICE_NAME="FlashGot Service";
  2257. const SERVICE_CID =
  2258.     Components.ID("{2a55fc5c-7b31-4ee1-ab15-5ee2eb428cbe}");
  2259. const SERVICE_CTRID =
  2260.     "@maone.net/flashgot-service;1";
  2261.     
  2262. const SERVICE_CONSTRUCTOR=FlashGotService;
  2263. const SERVICE_FLAGS = 3; // SINGLETON | THREADSAFE
  2264.  
  2265. // interfaces implemented by this component
  2266. const SERVICE_IIDS = 
  2267. [ 
  2268. Components.interfaces.nsISupports,
  2269. Components.interfaces.nsISupportsWeakReference,
  2270. Components.interfaces.nsIClassInfo,
  2271. Components.interfaces.nsIObserver,
  2272. Components.interfaces.nsIURIContentListener
  2273. ];
  2274.  
  2275. // Factory object
  2276. const SERVICE_FACTORY = {
  2277.   _instance: null,
  2278.   createInstance: function (outer, iid) {
  2279.     if (outer != null)
  2280.         throw Components.results.NS_ERROR_NO_AGGREGATION;
  2281.  
  2282.     xpcom_checkInterfaces(iid,SERVICE_IIDS,Components.results.NS_ERROR_INVALID_ARG);
  2283.     // kept this for flexibility sake, but we're really adopting an
  2284.     // early instantiation and late init singleton pattern
  2285.     return this._instance==null?this._instance=this._create():this._instance;
  2286.   },
  2287.   _create: function() {
  2288.     var obj=new SERVICE_CONSTRUCTOR();
  2289.     obj.__defineGetter__("classDescription",function() { return SERVICE_NAME; });
  2290.     obj.__defineGetter__("classID",function() { return SERVICE_CID; });
  2291.     obj.__defineGetter__("classIDNoAlloc",function() { return SERVICE_CTRID; });
  2292.     obj.__defineGetter__("contractID",function() { return SERVICE_CTRID; });
  2293.     obj.__defineGetter__("flags",function() { return SERVICE_FLAGS; });
  2294.     obj.__defineGetter__("implementationLanguage",function() { return 2; });
  2295.     obj.getHelperForLanguage = function() { return null; };
  2296.     obj.getInterfaces = function(count) { 
  2297.       count.value = 0; 
  2298.       return null; 
  2299.     };
  2300.     return obj;
  2301.   }
  2302. };
  2303.  
  2304. function xpcom_checkInterfaces(iid,iids,ex) {
  2305.   for(var j=iids.length; j-- >0;) {
  2306.     if(iid.equals(iids[j])) return true;
  2307.   }
  2308.   throw ex;
  2309. }
  2310.  
  2311. // Module
  2312.  
  2313. var Module = new Object();
  2314. Module.firstTime=true;
  2315. Module.registerSelf = function (compMgr, fileSpec, location, type) {
  2316.   if(this.firstTime) {
  2317.    
  2318.     debug("*** Registering "+SERVICE_CTRID+".\n");
  2319.     
  2320.     compMgr.QueryInterface(Components.interfaces.nsIComponentRegistrar
  2321.       ).registerFactoryLocation(SERVICE_CID,
  2322.       SERVICE_NAME,
  2323.       SERVICE_CTRID, 
  2324.       fileSpec,
  2325.       location, 
  2326.       type);
  2327.       
  2328.     Components.classes['@mozilla.org/categorymanager;1'].getService(
  2329.       Components.interfaces.nsICategoryManager
  2330.      ).addCategoryEntry("app-startup",
  2331.         SERVICE_NAME, "service," + SERVICE_CTRID, true, true, null);
  2332.       
  2333.     this.firstTime=false;
  2334.   } 
  2335. }
  2336. Module.unregisterSelf = function(compMgr, fileSpec, location) {
  2337.   compMgr.QueryInterface(Components.interfaces.nsIComponentRegistrar
  2338.     ).unregisterFactoryLocation(SERVICE_CID, fileSpec);
  2339.   Components.classes['@mozilla.org/categorymanager;1'].getService(
  2340.       Components.interfaces.nsICategoryManager
  2341.      ).deleteCategoryEntry("app-startup",SERVICE_NAME, true);
  2342. }
  2343.  
  2344. Module.getClassObject = function (compMgr, cid, iid) {
  2345.   if(cid.equals(SERVICE_CID))
  2346.     return SERVICE_FACTORY;
  2347.  
  2348.   if (!iid.equals(Components.interfaces.nsIFactory))
  2349.     throw Components.results.NS_ERROR_NOT_IMPLEMENTED;
  2350.   
  2351.   throw Components.results.NS_ERROR_NO_INTERFACE;
  2352.     
  2353. }
  2354.  
  2355. Module.canUnload = function(compMgr) {
  2356.   return true;
  2357. }
  2358.  
  2359. // entrypoint
  2360. function NSGetModule(compMgr, fileSpec) {
  2361.   return Module;
  2362. }
  2363.  
  2364.  
  2365.  
  2366.